package com.spzx.report.aiUtils;

import org.springframework.core.io.ClassPathResource;
import org.springframework.core.io.Resource;
import org.springframework.util.FileCopyUtils;

import java.io.IOException;
import java.io.InputStreamReader;
import java.time.LocalDateTime;

/*
* 作用: 提示词构造器
* 默认: 默认为空
* 系统: 自定义文本+系统模板文本
* */
public class PromptGenerator {
    //字符串提示词模板
    public static final String PROMPT_TEMPLATE = "";

    //文本提示词模板
    public static String systemPrompt = "你是一个精通电商领域的助手，需要根据用户的要求准确回答问题。"
            + PromptGenerator.readPromptFile();

    public static String readPromptFile() {
        try {
            Resource resource = new ClassPathResource("aiReportPrompt.txt");
            String content = FileCopyUtils.copyToString(new InputStreamReader(resource.getInputStream()));
            return content;
        } catch (IOException e) {
            e.printStackTrace();
        }
        return "";
    }

    public static String generatePrompt(String question){
        LocalDateTime now = LocalDateTime.now();
        return PROMPT_TEMPLATE + "\n[当前问题]" + question + "\n[当前时间]" + now;
    }
}
